package top.hkyzf.study.leetcode.linkedlist;

import org.junit.Test;

/**
 * @author 朱峰
 * @date 2022-2-16 11:30
 */
public class 移除链表元素_203 {
    public ListNode removeElements(ListNode head, int val) {
        ListNode reslut = new ListNode();
        reslut.next = head;
        ListNode cur = reslut;
        while (cur.next != null) {
            if (cur.next.val == val) {
                cur.next = cur.next.next;
            } else {
                cur = cur.next;
            }
        }
        return reslut.next;
    }

    @Test
    public void testRemoveElements () {
        ListNode node6 = new ListNode(6);
        ListNode node5 = new ListNode(5, node6);
        ListNode node4 = new ListNode(4, node5);
        ListNode node3 = new ListNode(6, node4);
        ListNode node2 = new ListNode(6, node3);
        ListNode node1 = new ListNode(2, node2);
        ListNode head = new ListNode(1, node1);
        int val = 6;
        ListNode node = removeElements(head, val);
        while (node != null) {
            System.out.println(node);
            node = node.next;
        }
    }
}
